Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import os
-from smolagents import CodeAgent, HfApiModel, tool
+from smolagents import CodeAgent, LiteLLMModel, tool
 import json
 from datetime import datetime
 import random
@@ -12,12 +12,22 @@ st.set_page_config(
     layout="wide"
 )
 
-# Initialize Hugging Face model
+# Initialize OpenAI model for Spaces
 @st.cache_resource
 def get_model():
-    return HfApiModel(model_id="HuggingFaceH4/zephyr-7b-beta")
+    api_key = os.environ.get("OPENAI_API_KEY")
+    if not api_key:
+        st.error("⚠️ OpenAI API Key not found in Spaces secrets!")
+        st.info("Add OPENAI_API_KEY to your Spaces secrets in Settings.")
+        st.stop()
+    
+    return LiteLLMModel(
+        model_id="gpt-4",
+        api_key=api_key,
+        temperature=0.2
+    )
 
-# Tool 1: Query SAP Purchase Requisition
+# [All your tool definitions remain exactly the same]
 @tool
 def query_sap_pr(pr_number: str) -> str:
     """
@@ -39,7 +49,6 @@ def query_sap_pr(pr_number: str) -> str:
     }
     return json.dumps(mock_data, indent=2)
 
-# Tool 2: Check Vendor Financial Health
 @tool  
 def check_vendor_solvency(vendor_name: str) -> str:
     """
@@ -61,7 +70,6 @@ def check_vendor_solvency(vendor_name: str) -> str:
     }
     return json.dumps(mock_data, indent=2)
 
-# Tool 3: Get Vendor Performance History
 @tool
 def get_vendor_performance(vendor_name: str) -> str:
     """
@@ -83,7 +91,6 @@ def get_vendor_performance(vendor_name: str) -> str:
     }
     return json.dumps(mock_data, indent=2)
 
-# Tool 4: Compare Material Rates
 @tool
 def compare_material_rates(material: str) -> str:
     """
@@ -113,7 +120,6 @@ def compare_material_rates(material: str) -> str:
     }
     return json.dumps(mock_data, indent=2)
 
-# Tool 5: Optimize Vendor Selection
 @tool
 def optimize_vendor_selection(requirements: str) -> str:
     """
@@ -137,7 +143,6 @@ def optimize_vendor_selection(requirements: str) -> str:
     }
     return json.dumps(mock_data, indent=2)
 
-# Tool 6: Create SAP Purchase Order
 @tool
 def create_purchase_order(vendor: str, amount: str) -> str:
     """
@@ -163,7 +168,7 @@ def create_purchase_order(vendor: str, amount: str) -> str:
     }
     return json.dumps(mock_data, indent=2)
 
-# Initialize Agent - FIXED: Removed max_iterations
+# Initialize Agent with OpenAI - FIXED: Removed max_iterations
 @st.cache_resource
 def get_agent():
     tools = [
@@ -175,18 +180,15 @@ def get_agent():
         create_purchase_order
     ]
     
-    # Fixed: No max_iterations parameter
     return CodeAgent(
         tools=tools,
-        model=get_model()
+        model=get_model()  # Using OpenAI via LiteLLM
     )
 
 def main():
-    # Header
     st.title("🤖 AI Procurement Agent")
-    st.markdown("*Demo powered by Hugging Face SmolAgents*")
+    st.markdown("*Demo powered by Hugging Face SmolAgents + OpenAI GPT-4*")
     
-    # Add info about the demo
     with st.expander("ℹ️ About this Demo"):
         st.markdown("""
         This AI agent can help automate procurement workflows by:
@@ -214,7 +216,7 @@ def main():
         )
         
         st.markdown("---")
-        st.markdown("**🚀 Deployed on Hugging Face Spaces**")
+        st.markdown("**🚀 Using OpenAI GPT-4**")
         
     # Main content
     col1, col2 = st.columns([1, 1])
@@ -262,8 +264,8 @@ def main():
                 with st.expander("📋 Full Agent Response", expanded=True):
                     st.markdown(st.session_state['agent_response'])
 
+# [Include all the workflow functions exactly as before]
 def process_full_workflow(pr_number):
-    """Process complete procurement workflow"""
     with st.spinner("🤖 AI Agent is working on your procurement request..."):
         prompt = f"""
         Execute a complete procurement workflow for PR {pr_number}:
@@ -287,7 +289,6 @@ def process_full_workflow(pr_number):
             st.error(f"Error: {str(e)}")
 
 def evaluate_vendor(vendor_name):
-    """Evaluate specific vendor"""
     with st.spinner(f"🤖 Evaluating {vendor_name}..."):
         prompt = f"""
         Conduct a comprehensive evaluation of vendor '{vendor_name}':
@@ -307,7 +308,6 @@ def evaluate_vendor(vendor_name):
             st.error(f"Error: {str(e)}")
 
 def analyze_prices(material):
-    """Analyze material prices"""
     with st.spinner(f"🤖 Analyzing prices for {material}..."):
         prompt = f"""
         Perform price analysis for '{material}':
@@ -327,7 +327,6 @@ def analyze_prices(material):
             st.error(f"Error: {str(e)}")
 
 def run_custom_query(query):
-    """Run custom user query"""
     with st.spinner("🤖 Processing your custom query..."):
         try:
             agent = get_agent()