Update app.py

app.py

@@ -100,23 +100,31 @@ async def api_home():
 # Endpoint to start a new conversation thread
 @app.post('/start_conversation/')
 async def start_conversation(request: Request):
-    data = await request.json()
-    prompt = data.get('prompt')
+    try:
+        data = await request.json()
+        prompt = data.get('prompt')
 
-    # Generate a response for the initial prompt
-    response = generate_response(prompt)
+        if not prompt:
+            raise HTTPException(status_code=400, detail="No prompt provided")
 
-    # Create a new conversation thread and store the prompt and response
-    thread_id = len(conversations) + 1
-    conversations[thread_id] = {'prompt': prompt, 'responses': [response]}
+        # Check if conversations dictionary is empty
+        if not conversations:
+            raise HTTPException(status_code=404, detail="No chat history available")
 
-    return {'thread_id': thread_id, 'response': response}
+        # Generate a response for the initial prompt
+        response = generate_response(prompt)
 
+        # Create a new conversation thread and store the prompt and response
+        thread_id = len(conversations) + 1
+        conversations[thread_id] = {'prompt': prompt, 'responses': [response]}
+
+        return {'thread_id': thread_id, 'response': response}
+    except HTTPException:
+        raise  # Re-raise HTTPException to return it directly
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
-    
-# Endpoint to get the response of a conversation thread
 
-    
 @app.get('/get_response/{thread_id}')
 async def get_response(thread_id: int):
     if thread_id not in conversations: