Enhance SaveInput model with Literal type for model_type and add confidence_score field

main.py

@@ -63,8 +63,9 @@ class SaveInput(BaseModel):
     username: str
     query: str
     retrieved_text: str
-    model_type: str
+    model_type: Literal["WhereIsAI_UAE_Large_V1", "BAAI_bge_large_en_v1.5"]
     reaction: str
+    confidence_score: float
 
 class SaveBatchInput(BaseModel):
     items: List[SaveInput]

static/index.html

@@ -210,51 +210,63 @@ class LoginResponse {
                     
                     <h4>cURL Request:</h4>
                     <pre><code class="language-bash">curl -X POST https://humblebeeai-al-ghazali-rag-retrieval-api.hf.space/save \
-     -H "Content-Type: application/json" \
-     -H "Authorization: Bearer your-access-token" \
-     -d '{
-           "items": [
-               {
-                   "user_type": "user",
-                   "username": "user123",
-                   "query": "test query",
-                   "retrieved_text": "Result 1 text",
-                   "model_type": "all-mpnet-base-v2",
-                   "reaction": "positive"
-               }
-           ]
-         }'</code></pre>
+                      -H "Content-Type: application/json" \
+                      -H "Authorization: Bearer YOUR_ACCESS_TOKEN" \
+                      -d '{
+                            "items": [
+                                {
+                                    "user_type": "user",
+                                    "username": "user1332",
+                                    "query": "What is the seventh test of lovers of God?",
+                                    "retrieved_text": "The seventh test is that lovers of God will love those who obey Him and hate the infidels and the disobedient...",
+                                    "model_type": "WhereIsAI/UAE-Large-V1",
+                                    "reaction": "positive",
+                                    "confidence_score": 0.95
+                                },
+                                {
+                                    "user_type": "user",
+                                    "username": "user1332",
+                                    "query": "What is the seventh test of lovers of God?",
+                                    "retrieved_text": "The seventh test is that lovers of God will love those who obey Him and hate the infidels and the disobedient...",
+                                    "model_type": "BAAI/bge-large-en-v1.5",
+                                    "reaction": "positive",
+                                    "confidence_score": 0.92
+                                }
+                            ]
+                          }'</code></pre>
 
                     <h4>Flutter Implementation:</h4>
                     <pre><code class="language-dart">class SaveInput {
-  final String userType;
-  final String username;
-  final String query;
-  final String retrievedText;
-  final String modelType;
-  final String reaction;
-
-  SaveInput({
-    required this.userType,
-    required this.username,
-    required this.query,
-    required this.retrievedText,
-    required this.modelType,
-    required this.reaction,
-  });
-
-  Map<String, dynamic> toJson() {
-    return {
-      'user_type': userType,
-      'username': username,
-      'query': query,
-      'retrieved_text': retrievedText,
-      'model_type': modelType,
-      'reaction': reaction,
-    };
-  }
-}
-
+                      final String userType;
+                      final String username;
+                      final String query;
+                      final String retrievedText;
+                      final String modelType;
+                      final String reaction;
+                      final double? confidenceScore;
+                    
+                      SaveInput({
+                        required this.userType,
+                        required this.username,
+                        required this.query,
+                        required this.retrievedText,
+                        required this.modelType,
+                        required this.reaction,
+                        this.confidenceScore,
+                      });
+                    
+                      Map<String, dynamic> toJson() {
+                        return {
+                          'user_type': userType,
+                          'username': username,
+                          'query': query,
+                          'retrieved_text': retrievedText,
+                          'model_type': modelType,
+                          'reaction': reaction,
+                          'confidence_score': confidenceScore,
+                        };
+                      }
+                    }
 Future<void> saveData(List<SaveInput> items, String accessToken) async {
   final url = Uri.parse('https://humblebeeai-al-ghazali-rag-retrieval-api.hf.space/save');