Update app.py

app.py

@@ -65,6 +65,33 @@ class SafeGeocoder:
             self.cache[location] = None
             return None
 
+# Function to just load the model
+def load_model():
+    try:
+        # Send a minimal request just to trigger model loading
+        payload = {
+            "inputs": "<|input|>\n### Template:\n{\"test\": \"\"}\n### Text:\ntest\n\n<|output|>",
+            "parameters": {
+                "max_new_tokens": 10,
+                "do_sample": False
+            }
+        }
+        
+        response = requests.post(API_URL, headers=headers, json=payload)
+        
+        if response.status_code == 503:
+            response_json = response.json()
+            if "error" in response_json and "loading" in response_json["error"]:
+                estimated_time = response_json.get("estimated_time", "unknown")
+                return f"⏳ Modell lädt... (ca. {int(float(estimated_time)) if isinstance(estimated_time, (int, float, str)) else 'unbekannt'} Sekunden)"
+        
+        if response.status_code != 200:
+            return f"❌ API Fehler: {response.status_code}"
+            
+        return "✅ Modell erfolgreich geladen! Sie können jetzt mit der Extraktion beginnen."
+    except Exception as e:
+        return f"❌ Fehler: {str(e)}"
+
 def extract_info(template, text):
     try:
         prompt = f"<|input|>\n### Template:\n{template}\n### Text:\n{text}\n\n<|output|>"
@@ -83,10 +110,10 @@ def extract_info(template, text):
             response_json = response.json()
             if "error" in response_json and "loading" in response_json["error"]:
                 estimated_time = response_json.get("estimated_time", "unknown")
-                return f"⏳ Model is loading (ETA: {int(float(estimated_time)) if isinstance(estimated_time, (int, float, str)) else 'unknown'} seconds)", "Please try again in a few minutes"
+                return f"⏳ Modell lädt... (ca. {int(float(estimated_time)) if isinstance(estimated_time, (int, float, str)) else 'unbekannt'} Sekunden)", "Bitte versuchen Sie es in einigen Minuten erneut oder nutzen Sie den 'Modell laden' Button"
         
         if response.status_code != 200:
-            return f"❌ API Error: {response.status_code}", response.text
+            return f"❌ API Fehler: {response.status_code}", response.text
         
         result = response.json()
         
@@ -104,11 +131,11 @@ def extract_info(template, text):
             extracted = json.loads(json_text)
             formatted = json.dumps(extracted, indent=2)
         except json.JSONDecodeError:
-            return "❌ JSON parsing error", json_text
+            return "❌ JSON Parsing Fehler", json_text
             
-        return "✅ Success", formatted
+        return "✅ Erfolgreich extrahiert", formatted
     except Exception as e:
-        return f"❌ Error: {str(e)}", "{}"
+        return f"❌ Fehler: {str(e)}", "{}"
 
 def create_map(df, location_col):
     m = folium.Map(
@@ -210,7 +237,7 @@ def process_excel(file, places_column):
         print(f"Spalten in der Excel-Tabelle: {list(df.columns)}")
         
         if places_column not in df.columns:
-            return None, f"Column '{places_column}' not found in the Excel file. Available columns: {', '.join(df.columns)}", None
+            return None, f"Spalte '{places_column}' wurde in der Excel-Datei nicht gefunden. Verfügbare Spalten: {', '.join(df.columns)}", None
         
         map_html, processed_count = create_map(df, places_column)
         
@@ -221,14 +248,14 @@ def process_excel(file, places_column):
         total_locations = df[places_column].count()
         success_rate = (processed_count / total_locations * 100) if total_locations > 0 else 0
         
-        stats = f"Found {processed_count} of {total_locations} locations ({success_rate:.1f}%)"
+        stats = f"Gefunden: {processed_count} von {total_locations} Orten ({success_rate:.1f}%)"
         
         return map_html, stats, processed_path
     except Exception as e:
         import traceback
         trace = traceback.format_exc()
         print(f"Error processing file: {e}\n{trace}")
-        return None, f"Error processing file: {str(e)}", None
+        return None, f"Fehler bei der Verarbeitung der Datei: {str(e)}", None
 
 custom_css = """
 <style>
@@ -258,6 +285,11 @@ h2 {
     background-color: #ff7518 !important;
 }
 
+.gradio-button.secondary {
+    background-color: #5a87ca !important;
+    color: white !important;
+}
+
 .info-box {
     background-color: #e8f4fd;
     border-left: 4px solid #2c6bb3;
@@ -294,6 +326,20 @@ h2 {
 .gr-box {
     margin-bottom: 0 !important;
 }
+
+/* Model status styling */
+.model-status {
+    padding: 10px;
+    border-radius: 4px;
+    margin-bottom: 15px;
+    background-color: #f8f9fa;
+    font-size: 14px;
+}
+
+.separator {
+    margin: 20px 0;
+    border-top: 1px solid #eaeaea;
+}
 </style>
 """
 
@@ -403,6 +449,24 @@ with gr.Blocks(css=custom_css, title="Daten Strukturieren und Analysieren") as d
             </div>
             """)
             
+            # Add model loading button and status at the top
+            with gr.Row():
+                load_btn = gr.Button("Modell laden (1. Schritt)", variant="secondary")
+                model_status = gr.Textbox(
+                    label="Modell Status", 
+                    value="Modell nicht geladen. Bitte zuerst das Modell laden, um Verzögerungen zu vermeiden.",
+                    elem_classes="model-status"
+                )
+            
+            # Connect loading button
+            load_btn.click(
+                fn=load_model,
+                inputs=[],
+                outputs=[model_status]
+            )
+            
+            gr.HTML("""<div class="separator"></div>""")
+            
             with gr.Row():
                 with gr.Column():
                     template = gr.Textbox(
@@ -415,7 +479,7 @@ with gr.Blocks(css=custom_css, title="Daten Strukturieren und Analysieren") as d
                         value="Nene Erdbeben in Japan. London, 15. Jan. (Drahtber.) Reuter meldet aus Osaka: Die telephonische Verbindung zwischen Osaka und Tokio ist heute um 5.45 Uhr durch ein Erdbeben unterbrochen worden. Die Straßenbahnen in Tokio liegen still. Der Eisenbahnverkehr Tokio — Osaka ist unterbrochen. Die kaiserliche Familie ist in Sicherheit. In Suvamo, einer Borstadt Tokios, sind Brände ausgebrochen. Ein Eisenbahnzug stürzte in den Bajubawo, einem Fluß zwischen Gotemba und Tokio. Sechs Züge wurden umgeworfen. Nenqork, 15. Jan. (Drahtber.) Aus Tokio wird berichtet, daß in Uokohama bei dem Erdbeben sechs Personen getötet und 22 verletzt wurden. In Tokio wurden vier Personen getötet und 20 verletzt. In Nokohama wurden 800 Häuser zerstört.",
                         lines=8
                     )
-                    extract_btn = gr.Button("Extrahieren Sie Informationen", variant="primary")
+                    extract_btn = gr.Button("Extrahieren Sie Informationen (2. Schritt)", variant="primary")
                 
                 with gr.Column():
                     status = gr.Textbox(label="Status")