Update app.py

app.py

@@ -223,13 +223,14 @@ import matplotlib.pyplot as plt
 from io import BytesIO
 from PIL import Image
 from simple_salesforce import Salesforce
+import gradio as gr
 
 # 📌 Salesforce Credentials
 SALESFORCE_USERNAME = "venkatramana@sandbox.com"
 SALESFORCE_PASSWORD = "Venkat12345@"
 SALESFORCE_SECURITY_TOKEN = "GhcJJmjBEefdnukJoz4CAQlR"
 
-# Function to Fetch and Display Salesforce Data
+# 📌 Function to Fetch and Display Salesforce Data (Stock Table & Graph)
 def fetch_salesforce_data():
     try:
         sf = Salesforce(
@@ -238,40 +239,76 @@ def fetch_salesforce_data():
             security_token=SALESFORCE_SECURITY_TOKEN
         )
 
+        # SOQL Query to fetch stock details
         query = "SELECT Product_Name__c, Current_Stocks__c FROM Inventory_Management__c LIMIT 100"
         response = sf.query(query)
         records = response.get('records', [])
 
         if not records:
-            return "<p>No Data Found</p>", None
+            return "<p>No Data Found</p>", None  # Return empty result if no records found
 
-        # Convert to DataFrame
+        # Convert Salesforce Data into Pandas DataFrame
         df = pd.DataFrame(records)
         df.rename(columns={'Product_Name__c': 'Product Name', 'Current_Stocks__c': 'Current Stocks'}, inplace=True)
         df.drop(columns=['attributes'], inplace=True, errors='ignore')
 
-        return df.to_html(index=False), generate_inventory_graph(df)
+        # Convert DataFrame to HTML Table
+        table_html = df.to_html(index=False)
+
+        # Generate Inventory Graph
+        graph_image = generate_inventory_graph(df)
+
+        return table_html, graph_image
 
     except Exception as e:
-        return f"Error fetching data: {str(e)}", None
+        return f"❌ Error fetching data: {str(e)}", None
 
-# Function to Generate Inventory Graph
+# 📊 Function to Generate Inventory Stock Graph
 def generate_inventory_graph(df):
     try:
         fig, ax = plt.subplots(figsize=(12, 6))
-        df.plot(kind='bar', x="Product Name", y="Current Stocks", ax=ax, legend=False)
-        ax.set_title("Inventory Stock Distribution")
-        ax.set_xlabel("Product Name")
-        ax.set_ylabel("Stock Quantity")
+        df.plot(kind='bar', x="Product Name", y="Current Stocks", ax=ax, legend=False, color="royalblue")
+        ax.set_title("Inventory Stock Distribution", fontsize=14, fontweight="bold")
+        ax.set_xlabel("Product Name", fontsize=12)
+        ax.set_ylabel("Stock Quantity", fontsize=12)
         plt.xticks(rotation=45, ha="right", fontsize=10)
         plt.tight_layout()
+
+        # Save the graph to a buffer
         buffer = BytesIO()
         plt.savefig(buffer, format="png")
         buffer.seek(0)
+
         return Image.open(buffer)
 
     except Exception as e:
-        return None
+        return None  # Return None if graph generation fails
+
+# 📌 Gradio UI for Viewing Salesforce Data (Table & Graph)
+def app():
+    with gr.Blocks() as interface:
+        gr.Markdown("<h1>📊 Salesforce Stock & Inventory Overview</h1>")
+
+        with gr.Row():
+            with gr.Column():
+                gr.Markdown("<h3>📋 Stock Data Table</h3>")
+                salesforce_table = gr.HTML(label="📦 Inventory Table")
+
+            with gr.Column():
+                gr.Markdown("<h3>📈 Inventory Stock Graph</h3>")
+                salesforce_graph = gr.Image(type="pil", label="📉 Stock Graph")
+
+        generate_button = gr.Button("⚡ Fetch Data")
+
+        # Clicking "Fetch Data" will display the table & graph
+        generate_button.click(fn=fetch_salesforce_data, inputs=[], outputs=[salesforce_table, salesforce_graph])
+
+    return interface
+
+# 🚀 Launch the Gradio App
+if __name__ == "__main__":
+    interface = app()
+    interface.launch(share=True)
 
 # Unified function to handle image processing and Salesforce interaction
 def process_image(image, mode, entry_type, quantity):