Update main.py

main.py

@@ -57,11 +57,8 @@ def setup_seeds(config):
 # ========================================
 
 SHARED_UI_WARNING = f'''### [NOTE] It is possible that you are waiting in a lengthy queue.
-
 You can duplicate and use it with a paid private GPU.
-
 <a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/Vision-CAIR/minigpt4?duplicate=true"><img style="margin-top:0;margin-bottom:0" src="https://huggingface.co/datasets/huggingface/badges/raw/main/duplicate-this-space-xl-dark.svg" alt="Duplicate Space"></a>
-
 Alternatively, you can also use the demo on our [project page](https://minigpt-4.github.io).
 '''
 
@@ -95,33 +92,54 @@ class Item(BaseModel):
     gr_img: UploadFile = File(..., description="Image file")
     text_input: str = None
 
+chat_state = CONV_VISION.copy()
+img_list = []
+chatbot = []
+
+
 @app.get("/")
 async def root():
     return RedirectResponse(url="/docs")
 
-@app.post("/process/")
-async def process_item(
+
+@app.post("/upload_img/")
+async def upload_img(
         file: UploadFile = File(...),
-        prompt: str = Form(...),
 ):
-    chat_state = CONV_VISION.copy()
-    img_list = []
-    chatbot=[]
     pil_image = Image.open(io.BytesIO(await file.read()))
     chat.upload_img(pil_image, chat_state, img_list)
+    return {"message": "image uploaded  successfully."}
+
+
+
+
+
+@app.post("/process/")
+async def process_item(
+        prompt: str = Form(...),
+):
+
+
     chat.ask(prompt, chat_state)
+    global chatbot
     chatbot = chatbot + [[prompt, None]]
     llm_message = chat.answer(conv=chat_state, img_list=img_list, max_new_tokens=300, num_beams=1, temperature=1,
                 max_length=2000)[0]
     chatbot[-1][1] = llm_message
     return chatbot
 
+@app.post("/reset/")
+async def reset(
 
-if __name__ == "__main__":
-    # Run the FastAPI app with Uvicorn
-    uvicorn.run("main:app", host="0.0.0.0", port=7860)
-
-
-
+):
+    global chat_state, img_list  # Use global keyword to reassign
+    img_list = []
+    if chat_state is not None:
+        chat_state.messages = []
+    if img_list is not None:
+        img_list = []
 
 
+if __name__ == "__main__":
+    # Run the FastAPI app with Uvicorn
+    uvicorn.run("main:app", host="0.0.0.0", port=7860)