Update app.py

app.py

@@ -28,7 +28,6 @@ def respond(
     if resume_file is not None:
         resume_text = extract_text_from_pdf(resume_file)
 
-    # Добавляем информацию о вакансии и резюме в системное сообщение
     system_message_with_info = f"{system_message}\nJob Description: {job_description}\nResume: {resume_text}"
     messages = [{"role": "system", "content": system_message_with_info}]
 
@@ -41,21 +40,25 @@ def respond(
     messages.append({"role": "user", "content": message})
 
     response = ""
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content
-        response += token
-        yield response
+    try:
+        for message in client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            if message.choices and message.choices[0].delta and message.choices[0].delta.content:
+                token = message.choices[0].delta.content
+                response += token
+                yield response
+            else:
+                yield "Error: received unexpected response format"
+    except Exception as e:
+        yield f"Error: {str(e)}"
 
-# Инициализация системного сообщения для старта
 initial_system_message = "You are a recruiter. Your task is to interview the candidate based on the job description and resume provided. After the interview, provide feedback on the candidate's strengths and areas for improvement."
 
-# Определение интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("## AI Recruiter Chatbot")
     
@@ -75,9 +78,12 @@ with gr.Blocks() as demo:
 
     def chat_flow(message, job_description, resume_file, max_tokens, temperature, top_p, history):
         system_message = initial_system_message
-        for response in respond(message, history, system_message, job_description, resume_file, max_tokens, temperature, top_p):
-            history.append((message, response))
-            yield response, history
+        response_generator = respond(message, history, system_message, job_description, resume_file, max_tokens, temperature, top_p)
+        final_response = ""
+        for response in response_generator:
+            final_response = response
+        history.append((message, final_response))
+        return final_response, history
 
     btn.click(chat_flow, [message, job_description, resume_file, max_tokens, temperature, top_p, history], [response, history])