Update app.py

app.py

@@ -20,7 +20,8 @@ OPENROUTER_API_URL = "https://openrouter.ai/api/v1/chat/completions"
 SUMMARIZER_MODELS = {
     "Default (facebook/bart-large-cnn)": "local_bart",
     "Free Model (distilbart-cnn-6-6)": "local_distilbart",
-    "OpenRouter (Claude-3)": "anthropic/claude-3-haiku"
+    "OpenRouter (Claude-3)": "anthropic/claude-3-haiku",
+    "OpenRouter (GPT-4)": "openai/gpt-4"
 }
 
 CACHE_SIZE = 500
@@ -100,19 +101,21 @@ def summarize_text(text, model_name):
 
         is_arabic = detect_language(text)
         
-        if is_arabic or model_name == "OpenRouter (Claude-3)":
-            logger.info("Using OpenRouter for summarization")
+        if is_arabic or model_name in ["OpenRouter (Claude-3)", "OpenRouter (GPT-4)"]:
+            logger.info(f"Using OpenRouter with model {model_name} for summarization")
             headers = {
                 "Authorization": f"Bearer {OPENROUTER_API_KEY}",
-                "HTTP-Referer": "https://localhost:7860",  # Replace with your actual domain
+                "HTTP-Referer": "https://localhost:7860",
                 "X-Title": "News Summarizer App",
                 "Content-Type": "application/json"
             }
             
             prompt = f"Please provide a concise summary of the following news article in the same language as the original text. Keep the summary brief and focused on key points:\n\n{text}"
             
+            model_id = SUMMARIZER_MODELS[model_name]
+            
             data = {
-                "model": "anthropic/claude-3-haiku",
+                "model": model_id,
                 "messages": [{"role": "user", "content": prompt}],
                 "max_tokens": 150
             }
@@ -224,7 +227,6 @@ def get_summary(tech_sources, business_sources, science_sources, world_sources,
         logger.error(f"Error in get_summary: {str(e)}")
         return f"An error occurred while processing your request: {str(e)}"
 
-# Gradio Interface
 # Gradio Interface
 demo = gr.Blocks()
 
@@ -278,7 +280,7 @@ with demo:
             )
     
     summarize_button = gr.Button("Get News Summary")
-    summary_output = gr.HTML(label="News Summary")  # Changed from gr.Textbox to gr.HTML
+    summary_output = gr.HTML(label="News Summary")
 
     summarize_button.click(
         get_summary,