Update app.py

app.py

@@ -1,36 +1,9 @@
 import gradio as gr
 import os
-# import subprocess
 import pandas as pd
 from PIL import Image
 import folium
 
-# --- Setup: Clone the repository for helper classes ---
-# This will only run once to fetch the necessary source code.
-# REPO_DIR = "TheAmateur"
-# if not os.path.exists(REPO_DIR):
-#     try:
-#         print(f"Cloning repository into ./{REPO_DIR}...")
-#         subprocess.run(
-#             ["git", "clone", "https://github.com/OrlovVladislav/TheAmateur.git"],
-#             check=True,
-#             capture_output=True,
-#             text=True
-#         )
-#         print("Repository cloned successfully.")
-#     except subprocess.CalledProcessError as e:
-#         print(f"Error cloning repository: {e.stderr}")
-#         exit() # Exit if cloning fails, as the app cannot run without it.
-
-# # Now that the repo is cloned, we can import the necessary modules
-# try:
-#     from TheAmateur.src.rainbolt_parody import RainboltParody
-#     from TheAmateur.src.map_gen import MapGenerator
-#     from TheAmateur.src.geo_info import GeoInfo
-# except ImportError as e:
-#     print(f"Failed to import from the cloned repository: {e}")
-#     exit()
-
 
 from src.rainbolt_parody import RainboltParody
 from src.map_gen import MapGenerator
@@ -102,18 +75,20 @@ def pinpoint_location(api_key, uploaded_image):
         reasoning = response.get('reasoning', 'No reasoning provided.')
         coords_str = response.get('coordinates', '0,0')
         
-        # --- FIX: Generate map HTML for the single point ---
         mgk = MapGenerator()
         single_point_map = mgk.get_map(location1=coords_str)
         map_html_output = single_point_map._repr_html_()
 
-        ui_updates = {
-            error_accordion: gr.Accordion(visible=True),
-            save_accordion: gr.Accordion(visible=True)
-        }
-        
-        # Return the map HTML instead of coordinates
-        return info_df, reasoning, map_html_output, response, ui_updates
+        # --- FIX IS HERE ---
+        # The return statement must have exactly 6 values to match the 6 outputs.
+        return (
+            info_df,                      # 1. for info_output_df
+            reasoning,                    # 2. for reasoning_output
+            map_html_output,              # 3. for map_output_single
+            response,                     # 4. for prediction_state
+            gr.Accordion(visible=True),   # 5. for error_accordion
+            gr.Accordion(visible=True)    # 6. for save_accordion
+        )
 
     except Exception as e:
         handle_error("Failed to process the image with the AI model", e)
@@ -211,7 +186,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="GeoGuessr AI") as app:
 
         with gr.Column(scale=2):
             gr.Markdown("### Map View")
-            # --- FIX: Changed gr.Map to gr.HTML ---
             map_output_single = gr.HTML(label="Predicted Location")
             map_output_comparison = gr.HTML(visible=False)