add font path feature

README.md

@@ -216,6 +216,7 @@ pepe_style_frog, crying with rain, emotional, dramatic lighting
 pepe-meme-generator/
 ├── src/                          # Source code
 │   ├── app.py                    # Main Streamlit application
+│   ├── Fonts/                    # Font files
 │   ├── model/                    # Model management
 │   │   ├── __init__.py
 │   │   ├── config.py             # Model configurations

src/app.py

@@ -256,6 +256,7 @@ def main():
         top_text = st.text_input("Top Text") if add_text else ""
         bottom_text = st.text_input("Bottom Text") if add_text else ""
         font_size = st.slider("Font Size", 10, 100, 40, 1)
+        font_path = config.FONT_PATH
     
     # Main area
     col1, col2 = st.columns([1, 1])
@@ -329,7 +330,7 @@ def main():
                 
                 # Add text if requested
                 if add_text and (top_text or bottom_text):
-                    image = processor.add_meme_text(image, top_text, bottom_text,font_size)
+                    image = processor.add_meme_text(image, top_text, bottom_text,font_size,font_path)
                 
                 # Always add MJ signature
                 image = processor.add_signature(image, signature="MJaheen", font_size=10, opacity=200)

src/model/config.py

@@ -111,6 +111,7 @@ class ModelConfig:
     LORA_PATH: str = "MJaheen/Pepe_The_Frog_model_v1_lora"
     USE_LORA: bool = True
     TRIGGER_WORD: str = "pepe_style_frog"
+    FONT_PATH: str = "src/Fonts/impact.ttf"
     
     # LCM settings
     USE_LCM: bool = False