Update app.py

app.py

@@ -45,14 +45,20 @@ def process_response(response, history):
     partial_words = "" 
     for chunk in response.iter_lines():
         if chunk:
-            chunk_data = json.loads(chunk[6:])['choices'][0]['delta']
-            if 'content' in chunk_data:
-                partial_words += chunk_data['content']
-                if token_counter == 0:
-                    history.append(" " + partial_words)
-                else:
-                    history[-1] = partial_words
-                token_counter += 1
+            try:
+                chunk_json = json.loads(chunk.decode('utf-8'))
+                if 'choices' in chunk_json and len(chunk_json['choices']) > 0:
+                    chunk_data = chunk_json['choices'][0].get('delta', {})
+                    if 'content' in chunk_data:
+                        partial_words += chunk_data['content']
+                        if token_counter == 0:
+                            history.append(" " + partial_words)
+                        else:
+                            history[-1] = partial_words
+                        token_counter += 1
+            except json.JSONDecodeError:
+                print("Error decoding JSON response")
+                break  # or handle the error as you see fit
     chat = [(history[i], history[i + 1]) for i in range(0, len(history) - 1, 2)]
     return chat, history, token_counter