Update app.py
Browse files
app.py
CHANGED
@@ -58,11 +58,8 @@ async def chat(request: ChatRequest):
|
|
58 |
{"role": "user", "parts": [request.message]},
|
59 |
]
|
60 |
|
61 |
-
|
62 |
-
|
63 |
-
)
|
64 |
-
if response and hasattr(response, 'text'):
|
65 |
-
return response.text # Return text if it's present
|
66 |
|
67 |
except Exception as e:
|
68 |
raise HTTPException(status_code=500, detail=str(e))
|
|
|
58 |
{"role": "user", "parts": [request.message]},
|
59 |
]
|
60 |
|
61 |
+
response = client.models.generate_content(model="gemini-2.0-flash", contents=request.message)
|
62 |
+
return response.text
|
|
|
|
|
|
|
63 |
|
64 |
except Exception as e:
|
65 |
raise HTTPException(status_code=500, detail=str(e))
|