Update app.py

app.py

@@ -1,11 +1,45 @@
+import gradio as gr
+import boto3
+import json
+import os
+from botocore.exceptions import ClientError, NoCredentialsError
+
+# Configuration - Set via Hugging Face Spaces environment variables
+KNOWLEDGE_BASE_ID = os.environ.get("KNOWLEDGE_BASE_ID", "PLEASE_SET_IN_SPACES_SETTINGS")
+AWS_REGION = os.environ.get("AWS_REGION", "us-east-1")
+MODEL_ARN = os.environ.get("MODEL_ARN", "arn:aws:bedrock:us-east-1::foundation-model/amazon.titan-text-express-v1")
+
+# Check if running in demo mode (no credentials set)
+DEMO_MODE = KNOWLEDGE_BASE_ID == "PLEASE_SET_IN_SPACES_SETTINGS"
+
+# Initialize AWS clients
+bedrock_agent = None
+initialization_error = None
+
+if not DEMO_MODE:
+    try:
+        bedrock_agent = boto3.client(
+            'bedrock-agent-runtime',
+            region_name=AWS_REGION,
+            aws_access_key_id=os.environ.get('AWS_ACCESS_KEY_ID'),
+            aws_secret_access_key=os.environ.get('AWS_SECRET_ACCESS_KEY')
+        )
+        print(f"✅ AWS Bedrock client initialized in region: {AWS_REGION}")
+    except Exception as e:
+        print(f"❌ Error initializing AWS client: {e}")
+        initialization_error = str(e)
+else:
+    print("🚧 Running in DEMO MODE - Configure environment variables to enable AWS integration")
+
+
 def search_knowledge_base(query, search_type="generate"):
     """
     Search the ORU IT Knowledge Base using Amazon Bedrock
     """
-    
+
     if not query.strip():
         return "Please enter a question!", "", "error"
-    
+
     # Demo mode response
     if DEMO_MODE:
         demo_response = f"""
@@ -19,34 +53,26 @@ To enable full functionality:
 1. Configure your Amazon Bedrock Knowledge Base
 2. Set up environment variables in Hugging Face Spaces settings
 3. Upload your ORU IT documentation to the knowledge base
-
-**Expected Response:** Once configured, this system would search through ORU's IT knowledge base and provide accurate, contextual answers to questions like:
-- Password reset procedures
-- Software installation guides
-- Network troubleshooting steps
-- Campus system access instructions
         """
-        
+
         demo_sources = """
 **Demo Sources:**
 - password_authentication_001_student_password_change.txt
 - mobile_applications_002_oru_app_experience_fix.txt  
 - software_applications_003_word_mac_view_only_fix.txt
-
-*These would be the actual ORU IT knowledge base documents that contain the answer.*
         """
-        
+
         return demo_response, demo_sources, "demo"
-    
+
     if not bedrock_agent:
         error_msg = f"❌ AWS Bedrock client not initialized."
         if initialization_error:
             error_msg += f" Error: {initialization_error}"
         return error_msg, "", "error"
-    
+
     try:
         if search_type == "generate":
-            # ✅ Corrected RetrieveAndGenerate with separate modelConfiguration
+            # ✅ Correct RetrieveAndGenerate with separate modelConfiguration
             response = bedrock_agent.retrieve_and_generate(
                 input={'text': query},
                 retrieveAndGenerateConfiguration={
@@ -64,27 +90,28 @@ To enable full functionality:
                     }
                 }
             )
-            
+
             # Extract the generated response
             generated_text = response.get('output', {}).get('text', 'No response generated.')
-            
+
             # Extract sources/citations
             sources_info = []
             citations = response.get('citations', [])
-            
+
             for i, citation in enumerate(citations, 1):
                 retrieved_refs = citation.get('retrievedReferences', [])
                 for j, ref in enumerate(retrieved_refs, 1):
                     content = ref.get('content', {}).get('text', 'No content available')
                     location = ref.get('location', {}).get('s3Location', {}).get('uri', 'Unknown source')
-                    sources_info.append(f"**Source {i}.{j}:**\n{content[:300]}{'...' if len(content) > 300 else ''}\n*From: {location}*\n")
-            
+                    sources_info.append(
+                        f"**Source {i}.{j}:**\n{content[:300]}{'...' if len(content) > 300 else ''}\n*From: {location}*\n"
+                    )
+
             sources_text = "\n".join(sources_info) if sources_info else "No sources found."
-            
+
             return generated_text, sources_text, "success"
-            
+
         else:  # retrieve only
-            # Use Retrieve for sources only
             response = bedrock_agent.retrieve(
                 knowledgeBaseId=KNOWLEDGE_BASE_ID,
                 retrievalQuery={'text': query},
@@ -94,34 +121,132 @@ To enable full functionality:
                     }
                 }
             )
-            
+
             retrieval_results = response.get('retrievalResults', [])
             if not retrieval_results:
                 return "No relevant sources found.", "", "warning"
-            
+
             sources_info = []
             for i, result in enumerate(retrieval_results, 1):
                 content = result.get('content', {}).get('text', 'No content available')
                 score = result.get('score', 0)
                 location = result.get('location', {}).get('s3Location', {}).get('uri', 'Unknown source')
+
                 sources_info.append(
                     f"**Source {i} (Confidence: {score:.2f}):**\n"
                     f"{content[:400]}{'...' if len(content) > 400 else ''}\n"
                     f"*From: {location}*\n"
                 )
-            
+
             sources_text = "\n".join(sources_info)
             response_text = f"Found {len(retrieval_results)} relevant sources from the ORU IT Knowledge Base:"
-            
+
             return response_text, sources_text, "success"
-            
+
     except ClientError as e:
         error_code = e.response['Error']['Code']
         error_message = e.response['Error']['Message']
         return f"❌ AWS Error ({error_code}): {error_message}", "", "error"
-        
+
     except NoCredentialsError:
         return "❌ AWS credentials not found. Please configure your credentials in Spaces settings.", "", "error"
-        
+
     except Exception as e:
         return f"❌ Unexpected error: {str(e)}", "", "error"
+
+
+def format_response(query, search_type):
+    """Process the query and return formatted response"""
+
+    if not query.strip():
+        return (
+            "Please enter your IT question above and click 'Get AI Answer' or 'Show Sources'.",
+            "",
+            "Enter a question to get started!"
+        )
+
+    response_text, sources_text, status = search_knowledge_base(query, search_type)
+
+    if status == "demo":
+        formatted_response = response_text
+        status_msg = "🚧 Demo Mode - Configure AWS credentials to enable full functionality"
+    elif search_type == "generate":
+        if status == "success":
+            formatted_response = f"## 🤖 AI Assistant Response\n\n{response_text}"
+            status_msg = "✅ Response generated successfully from ORU IT Knowledge Base!"
+        else:
+            formatted_response = response_text
+            status_msg = "❌ Error generating response"
+    else:
+        if status == "success":
+            formatted_response = f"## 📚 Retrieved Sources\n\n{response_text}"
+            status_msg = "✅ Sources retrieved successfully from ORU IT Knowledge Base!"
+        else:
+            formatted_response = response_text
+            status_msg = "❌ Error retrieving sources"
+
+    return formatted_response, sources_text, status_msg
+
+
+# Sample queries
+SAMPLE_QUERIES = [
+    "How do I change my student password?",
+    "My ORU app isn't showing the right tiles",
+    "Word is showing view only mode on my Mac",
+    "How do I set up multi-factor authentication?",
+    "I can't edit Microsoft Word documents",
+    "How do I connect to ORU WiFi network?",
+    "My ORU email isn't syncing to my phone",
+    "How do I access Vision portal from off-campus?",
+    "VPN connection keeps dropping",
+    "Can't print to campus printers from my laptop",
+]
+
+# Build Gradio UI
+with gr.Blocks(title="ORU IT Helpdesk Assistant") as app:
+    gr.Markdown("## 🏛️ Oral Roberts University IT Helpdesk Assistant\nPowered by Amazon Bedrock Knowledge Bases & S3 Vectors")
+
+    with gr.Row():
+        with gr.Column(scale=2):
+            query_input = gr.Textbox(label="🔍 Ask your IT question:")
+            with gr.Row():
+                search_btn = gr.Button("🤖 Get AI Answer", variant="primary")
+                retrieve_btn = gr.Button("📚 Show Sources", variant="secondary")
+
+            with gr.Column():
+                for q in SAMPLE_QUERIES:
+                    btn = gr.Button(q, size="sm")
+                    btn.click(lambda x=q: x, outputs=query_input)
+
+        with gr.Column(scale=3):
+            status_output = gr.HTML()
+            response_output = gr.Markdown()
+            sources_output = gr.Markdown(visible=True)
+
+    search_btn.click(
+        fn=lambda query: format_response(query, "generate"),
+        inputs=[query_input],
+        outputs=[response_output, sources_output, status_output]
+    )
+    retrieve_btn.click(
+        fn=lambda query: format_response(query, "retrieve"),
+        inputs=[query_input],
+        outputs=[response_output, sources_output, status_output]
+    )
+    query_input.submit(
+        fn=lambda query: format_response(query, "generate"),
+        inputs=[query_input],
+        outputs=[response_output, sources_output, status_output]
+    )
+
+
+# ✅ Always launch on Hugging Face
+print("🏛️ Starting ORU IT Helpdesk Assistant...")
+print(f"📚 Knowledge Base ID: {KNOWLEDGE_BASE_ID}")
+print(f"🌍 AWS Region: {AWS_REGION}")
+print(f"🚧 Demo Mode: {'Active' if DEMO_MODE else 'Disabled'}")
+
+app.launch(
+    server_name="0.0.0.0",
+    server_port=int(os.environ.get("PORT", 7860))
+)