Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import torch
 from transformers import MarianMTModel, MarianTokenizer
 import logging
-import os
+import json
 
 # ---------------- LOGGING ----------------
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
@@ -29,7 +29,7 @@ class FrenchTitleGenerator:
             original_title = str(original_title).strip()
             
             if not original_title:
-                return "Please enter an event description."
+                return {"error": "Please enter an event description."}
             
             input_text = f"Translate to French: {original_title}"
             
@@ -53,19 +53,31 @@ class FrenchTitleGenerator:
             
             # Add context if needed
             if year and city:
-                return f"{result} - {city} {year}"
+                final_result = f"{result} - {city} {year}"
             elif city:
-                return f"{result} - {city}"
+                final_result = f"{result} - {city}"
             elif year:
-                return f"{result} - {year}"
+                final_result = f"{result} - {year}"
             else:
-                return result
+                final_result = result
+
+            return {
+                "success": True,
+                "french_title": final_result,
+                "input_used": {
+                    "year": year,
+                    "city": city,
+                    "original_title": original_title,
+                    "temperature": temperature,
+                    "max_length": max_length
+                }
+            }
 
         except Exception as e:
             logger.error(f"Error during generation: {e}")
-            return f"Error generating title: {str(e)}"
+            return {"error": f"Error generating title: {str(e)}"}
 
-# ---------------- INIT MODEL WITH ERROR HANDLING ----------------
+# ---------------- INIT MODEL ----------------
 try:
     generator = FrenchTitleGenerator()
     model_loaded = True
@@ -80,12 +92,14 @@ def generate_title(year, city, original_title, temperature=0.8, max_length=50):
     if not model_loaded:
         return "Error: Model failed to load. Please check the logs."
     
-    if not original_title.strip():
-        return "Please enter an event description."
+    result = generator.generate(year, city, original_title, max_length, temperature)
     
-    return generator.generate(year, city, original_title, max_length=max_length, temperature=temperature)
+    if "error" in result:
+        return result["error"]
+    else:
+        return result["french_title"]
 
-# Create simple interface without advanced features
+# Create interface
 iface = gr.Interface(
     fn=generate_title,
     inputs=[
@@ -104,9 +118,13 @@ iface = gr.Interface(
     ),
     title="🇫🇷 French Event Title Generator",
     description="Generate elegant French titles for your events based on year, city, and description.",
-    allow_flagging="never"
+    allow_flagging="never",
+    api_name="predict"
 )
 
 if __name__ == "__main__":
-    # Simple launch without complex options
-    iface.launch(server_name="0.0.0.0", server_port=7860)
+    iface.launch(
+        server_name="0.0.0.0", 
+        server_port=7860,
+        share=False
+    )