Upload app.py with huggingface_hub

app.py

@@ -1,35 +1,33 @@
 import gradio as gr
 from llama_cpp import Llama
 from huggingface_hub import hf_hub_download
+import os
 
 try:
-    print("--- Downloading Model from Model Hub ---")
+    # تحميل ملف GGUF من المستودع الخارجي لتجاوز حد المساحة (1GB)
+    print("--- Downloading Gemma 4 GGUF Model ---")
     model_path = hf_hub_download(
         repo_id='Arabic250/gemma-4-gguf-export',
         filename='gemma-4-medical.gguf'
     )
 
-    print("--- Initializing Llama-CPP ---")
-    llm = Llama(model_path=model_path, n_ctx=2048, n_threads=2)
+    # تهيئة النموذج
+    print("--- Initializing Engine ---")
+    llm = Llama(model_path=model_path, n_ctx=1024, n_threads=2)
 
-    def chat(message, history):
-        # بناء سياق المحادثة ليحتفظ بالرسائل السابقة
-        prompt = ""
-        for user_msg, bot_msg in history:
-            prompt += f"USER: {user_msg}\nASSISTANT: {bot_msg}\n"
-        
-        prompt += f"USER: {message}\nASSISTANT: "
-
-        # echo=False تمنع النموذج من تكرار السؤال في الإجابة
-        output = llm(prompt, max_tokens=512, stop=['USER:'], echo=False)
-        return output['choices'][0]['text']
-
-    demo = gr.ChatInterface(fn=chat, title='Gemma 4 Medical')
+    def predict(message, history):
+        prompt = f"USER: {message}\nASSISTANT: "
+        response = llm(prompt, max_tokens=512, stop=['USER:'], echo=False)
+        return response['choices'][0]['text']
 
+    demo = gr.ChatInterface(
+        fn=predict, 
+        title='⚡ Gemma 4 Medical - Docker Edition',
+        description='واجهة دردشة طبية تعمل بنظام Docker المستقر.'
+    )
 except Exception as e:
     with gr.Blocks() as demo:
-        gr.Markdown(f'# Error: {e}')
+        gr.Markdown(f"# ❌ Error: {e}")
 
 if __name__ == '__main__':
-    # تحديد المنفذ 7860 صراحةً وهو المنفذ الافتراضي الذي تستمع إليه Hugging Face Spaces
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name='0.0.0.0', server_port=7860)