Refined rizz

api/routers/reply_generator.py

@@ -13,7 +13,6 @@ graph = reply_generator.run()
 
 class UserRequest(BaseModel):
     image_base64: str
-    user_query: Optional[str] = None
     tones: Optional[List[str]] = None
 
 @router.post("/reply-generator")
@@ -27,7 +26,6 @@ async def generate_reply(request: UserRequest):
     stored_data['conversation_context'] = conversation_text
 
     result = graph.invoke({
-        'messages': [request.user_query],
         'tones': request.tones,
         'conversation_chat': stored_data['conversation_context']
     },config=config)

src/pickup_line_generator/generate.py

@@ -12,8 +12,8 @@ class Generate:
     
     def run(self, state:State):
         messages = [SystemMessage(content=pickup_line_prompt),
-        FunctionMessage(name='attrubutes',content=f'''The attributes of person are: {state['attributes']}'''),
-         HumanMessage(content=f'''The required tone: {state["tones"]}''')]
+        FunctionMessage(name='attrubutes',content=f'''The attributes of person are: {state['attributes']}\n. The user is saying: {state['messages'][-1]}'''),
+        HumanMessage(content=f'''The required tone: {state["tones"]}''')]
         pickup_lines = reply_llm.with_structured_output(PickupFormatter).invoke(messages)
         return {
             'messages':[{'role': 'assistant', 'content': str(pickup_lines.model_dump())}],

src/reply_generator/generate.py

@@ -19,7 +19,7 @@ class Generate:
         print(reply.model_dump())
         
         return {
-            'messages':[{'role': 'assistant', 'content': str(reply.model_dump())}],
+            'messages':[{'role': 'assistant', 'content': f'''The generated replies are: {str(reply.model_dump())}'''}],
             'replies': reply.model_dump()
         }
         

streamlit/pages/check_looks.py

@@ -3,7 +3,9 @@ import requests
 import base64
 
 # FastAPI endpoint
-API_URL = "https://dvorakinnovationai-revel-rizz-api.hf.space/api/looks-analyzer"  # Change if deployed elsewhere
+API_URL = "https://dvorakinnovationai-revel-rizz-api.hf.space/api/looks-analyzer" 
+# API_URL = "http://127.0.0.1:8000//api/looks-analyzer"
+
 
 st.set_page_config(page_title="Looks Analyzer", page_icon="🧑‍🎨", layout="centered")
 

streamlit/pages/pickup_line_generator.py

@@ -2,7 +2,9 @@ import streamlit as st
 import requests
 
 # FastAPI endpoint
-API_URL = "https://dvorakinnovationai-revel-rizz-api.hf.space/api/pickup-line-generator"  # Change if deployed elsewhere
+API_URL = "https://dvorakinnovationai-revel-rizz-api.hf.space/api/pickup-line-generator"
+# API_URL = "http://127.0.0.1:8000//api/pickup-line-generator"
+
 
 st.set_page_config(page_title="Pickup Line Generator", page_icon="💘", layout="centered")
 
@@ -32,7 +34,7 @@ if st.button("Generate Pickup Line"):
             attributes_list = [attr.strip() for attr in attributes.split(",") if attr.strip()]
 
             payload = {
-                "user_query": user_query if user_query.strip() else None,
+                "user_query": user_query if user_query.strip() else '',
                 "tones": tones if tones else None,
                 "attributes": attributes_list,
             }

streamlit/pages/reply_generator.py

@@ -3,8 +3,8 @@ import requests
 import base64
 
 # FastAPI endpoint
-API_URL = "https://dvorakinnovationai-revel-rizz-api.hf.space/api/reply-generator"  # Change if deployed elsewhere
-
+API_URL = "https://dvorakinnovationai-revel-rizz-api.hf.space/api/reply-generator" 
+# API_URL = "http://127.0.0.1:8000//api/reply-generator"
 
 st.set_page_config(page_title="Reply Generator", page_icon="💬", layout="centered")
 
@@ -15,7 +15,6 @@ st.write("Upload image of chat , ask a question, and get AI-powered replies.")
 uploaded_file = st.file_uploader("Upload an image (JPG/PNG)", type=["jpg", "jpeg", "png"])
 
 # User query input
-user_query = st.text_area("Enter your query", placeholder="Type your question here...")
 
 # Tones (multi-select)
 available_tones = ["formal", "casual", "friendly", "professional", "humorous"]
@@ -25,8 +24,7 @@ tones = st.multiselect("Select tones (optional)", available_tones)
 if st.button("Generate Reply"):
     if uploaded_file is None:
         st.error("⚠️ Please upload an image first.")
-    elif not user_query.strip():
-        st.error("⚠️ Please enter a query.")
+
     else:
         try:
             # Convert image to base64
@@ -35,7 +33,6 @@ if st.button("Generate Reply"):
 
             payload = {
                 "image_base64": image_base64,
-                "user_query": user_query,
                 "tones": tones if tones else None,
             }