Update app.py

app.py

@@ -116,46 +116,6 @@ img_interface = gr.Interface(
     title="AI Image Question Answering"
 )
 
-# ✅ Data Visualization Function
-def generate_visualization(excel_file, viz_type, user_request):
-    try:
-        df = pd.read_excel(excel_file)
-        df = df.astype(str).fillna("")
-
-        table_input = {
-            "table": df.to_dict(orient="records"),  
-            "query": user_request.strip() if isinstance(user_request, str) else "What is the summary?"
-        }
-
-        table_answer = table_analyzer(**table_input)  
-
-        prompt = (
-            f"Given a dataset with columns {list(df.columns)}, generate Python code using Matplotlib and Seaborn "
-            f"to create a {viz_type.lower()} based on: {user_request}. Only return valid Python code, no explanations."
-        )
-        code_response = code_generator(prompt, max_new_tokens=150, do_sample=True)
-
-        if isinstance(code_response, list) and "generated_text" in code_response[0]:
-            generated_code = code_response[0]["generated_text"]
-        else:
-            generated_code = "Error: Model did not return valid code."
-
-        try:
-            exec_globals = {"plt": plt, "sns": sns, "pd": pd, "df": df, "io": io}
-            exec(generated_code, exec_globals)
-            
-            fig = plt.gcf()
-            img_buf = io.BytesIO()
-            fig.savefig(img_buf, format='png')
-            img_buf.seek(0)
-            plt.close(fig)  
-        except Exception as e:
-            return generated_code, f"Error in executing visualization: {str(e)}"
-        
-        return generated_code, img_buf
-
-    except Exception as e:
-        return f"Error: {str(e)}", "Failed to analyze table."
 
 # ✅ Gradio UI for Data Visualization
 viz_interface = gr.Interface(
@@ -170,7 +130,7 @@ viz_interface = gr.Interface(
 )
 
 # ✅ Mount Gradio Interfaces
-demo = gr.TabbedInterface([doc_interface, img_interface, viz_interface], ["Document QA", "Image QA"])
+demo = gr.TabbedInterface([doc_interface, img_interface], ["Document QA", "Image QA"])
 app = gr.mount_gradio_app(app, demo, path="/")
 
 @app.get("/")