Update app.py

app.py

@@ -1,15 +1,14 @@
 # app.py
 """
-Main application file for SHASHA AI, a Gradio‑based AI code‑generation tool.
-
-Only change: reduce logo width so the banner isn’t full‑width.
+Main application file for SHASHA AI (Gradio).  
+Only change: enlarge logo to 120 × 120 px.
 """
 
 import gradio as gr
 from typing import Optional, Dict, List, Tuple, Any
 import os
 
-# --- Local module imports ---
+# --- Local imports ---
 from constants import (
     HTML_SYSTEM_PROMPT,
     TRANSFORMERS_JS_SYSTEM_PROMPT,
@@ -30,7 +29,6 @@ from utils import (
 )
 from deploy import send_to_sandbox
 
-# --- Aliases ---
 History = List[Tuple[str, str]]
 Model   = Dict[str, Any]
 
@@ -63,15 +61,16 @@ def generation_code(
         )
         model_id = current_model["id"]
         provider = (
-            "openai"        if model_id.startswith("openai/") or model_id in {"gpt-4","gpt-3.5-turbo"}
-            else "gemini"   if model_id.startswith(("gemini/","google/"))
+            "openai" if model_id.startswith("openai/") or model_id in {"gpt-4","gpt-3.5-turbo"}
+            else "gemini" if model_id.startswith(("gemini/","google/"))
             else "fireworks-ai" if model_id.startswith("fireworks-ai/")
             else "auto"
         )
 
         msgs    = history_to_messages(history, system_prompt)
         context = query
-        if file:        context += f"\n\n[Attached file]\n{extract_text_from_file(file)[:5000]}"
+        if file:
+            context += f"\n\n[Attached file]\n{extract_text_from_file(file)[:5000]}"
         if website_url:
             wtext = extract_website_content(website_url)
             if not wtext.startswith("Error"):
@@ -79,7 +78,7 @@ def generation_code(
         msgs.append({"role":"user","content":enhance_query_with_search(context, enable_search)})
 
         client  = get_inference_client(model_id, provider)
-        resp    = client.chat.completions.create(model=model_id, messages=msgs, max_tokens=16000, temperature=0.1)
+        resp    = client.chat.completions.create(model=model_id, messages=msgs,max_tokens=16000,temperature=0.1)
         content = resp.choices[0].message.content
 
     except Exception as e:
@@ -117,10 +116,10 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"),
     initial_model = AVAILABLE_MODELS[0]
     model_state   = gr.State(initial_model)
 
+    # enlarged logo (120×120 px)
     if os.path.exists(LOGO_PATH):
-        gr.Image(value=LOGO_PATH, height=70, width=70,
-                 show_label=False, container=False,
-                 elem_id="banner_logo")
+        gr.Image(value=LOGO_PATH, height=120, width=120,
+                 show_label=False, container=False)
 
     gr.Markdown("# 🚀 Shasha AI", elem_id="main_title")
     gr.Markdown("Your AI partner for generating, modifying, and understanding code.", elem_id="subtitle")